package com.example.his.api.db.pojo;

import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Repository;

import javax.annotation.Resource;

/**
 * @author huixh
 */
@Repository
public class GoodsSnapshotDao {

    @Resource
    private MongoTemplate mongoTemplate;

    public String hashGoodsSnapshot(String md5) {
        Criteria criteria = Criteria.where("md5").is(md5);
        Query query = new Query(criteria);
        query.skip(0);
        query.limit(1);
        GoodsSnapshotEntity snapshotEntity = mongoTemplate.findOne(query, GoodsSnapshotEntity.class);
        return snapshotEntity != null ? snapshotEntity.get_id() : null;
    }

    public String insertGoodsSnapshot(GoodsSnapshotEntity goodsSnapshotEntity) {
        GoodsSnapshotEntity snapshotEntity = mongoTemplate.insert(goodsSnapshotEntity);
        return snapshotEntity.get_id();
    }
}
